/**
 * 给你一个链表，每 k 个节点一组进行翻转，请你返回翻转后的链表。
 * <p>
 * k 是一个正整数，它的值小于或等于链表的长度。
 * <p>
 * 如果节点总数不是 k 的整数倍，那么请将最后剩余的节点保持原有顺序。
 * <p>
 *  
 * <p>
 * 示例：
 * <p>
 * 给你这个链表：1->2->3->4->5
 * <p>
 * 当 k = 2 时，应当返回: 2->1->4->3->5
 * <p>
 * 当 k = 3 时，应当返回: 3->2->1->4->5
 * <p>
 *  
 * <p>
 * 说明：
 * <p>
 * 你的算法只能使用常数的额外空间。
 * 你不能只是单纯的改变节点内部的值，而是需要实际进行节点交换。
 * <p>
 * Definition for singly-linked list.
 * public class ListNode {
 * int val;
 * ListNode next;
 * ListNode() {}
 * ListNode(int val) { this.val = val; }
 * ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {

    public static void main(String[] args) {
        ListNode node1 = new ListNode(1);
        ListNode node2 = new ListNode(2);
        ListNode node3 = new ListNode(3);
        ListNode node4 = new ListNode(4);
        ListNode node5 = new ListNode(5);
        ListNode node6 = new ListNode(6);
        ListNode node7 = new ListNode(7);
        node1.next = node2;
        node2.next = node3;
        node3.next = node4;
        node4.next = node5;
        node5.next = node6;
        node6.next = node7;
        ListNode node = reverseKGroup(node1, 3);
        System.out.println(node.val);
        while (node.next != null) {
            node = node.next;
            System.out.println(node.val);
        }
    }

    /**
     * 特殊情况特殊考虑
     * @param head
     * @param k
     * @return
     */
    public static ListNode reverseKGroup(ListNode head, int k) {
        if (head == null || k <= 1) {
            return head;
        } else if (k == 2) {
            return swapPairs(head);
        } else {
            return swapMore(head, k);
        }
    }

    /**
     * 校验两个
     *
     * @param head
     * @return
     */
    public static ListNode swapPairs(ListNode head) {
        if (head != null && head.next != null) {
            ListNode newHead = head.next;
            // 如果当前节点存在后续节点，则交换
            if (head.next.next != null) {
                // 如果当前节点的后续节点还存在后续节点，则继续交换
                ListNode tmp = swapPairs(head.next.next);
                newHead.next = head;
                newHead.next.next = tmp;
            } else {
                newHead.next = head;
                newHead.next.next = null;
            }
            return newHead;
        } else {
            // 返回当前节点
            return head;
        }
    }

    /**
     * 交换k个
     *
     * @param head
     * @param k
     * @return
     */
    public static ListNode swapMore(ListNode head, int k) {
        if (head == null) {
            return null;
        }
        // 往后找第k个元素是不是存在
        ListNode lastNode = head;
        for (int i = 0; i < k - 1; i++) {
            if (lastNode.next == null) {
                // 不存在则返回
                return head;
            } else {
                lastNode = lastNode.next;
            }
        }
        ListNode left = swapMore(lastNode.next, k);
        // 如果存在则更换箭头方向
        ListNode pre = head;
        ListNode next = head.next;
        ListNode newPre, newNext;
        for (int i = 0; i < k - 1; i++) {
            newPre = next;
            newNext = next.next;
            next.next = pre;
            pre = newPre;
            next = newNext;
        }
        head.next = left;
        return pre;
    }

}

class ListNode {
    int val;
    ListNode next;

    ListNode() {
    }

    ListNode(int val) {
        this.val = val;
    }

    ListNode(int val, ListNode next) {
        this.val = val;
        this.next = next;
    }
}